#!/bin/sh
#SBATCH --job-name=case20.1
#SBATCH -N 1
#SBATCH -n 14
#SBATCH --gres=gpu:1 ## Run on 1 GPU
#SBATCH --output job%j.out
#SBATCH --error job%j.err
#SBATCH -p v100-32gb-hiprio


##Load your modules and run code here

#hostname
#
module load cuda/11.1
module load python3/anaconda/2021.11
#
source activate
#conda activate /work/ygeng/ENVS
cd /work/ygeng/case2_V5
python ./train_model.py --height 128 --width 128 --intial_snap 1 --mid_channels 16 --deltaT 0.1 --trainingBatch 50 --testBatch 1 --type 'BE' --epsilon 0.01 --TotalSnaps 10 --TotalTrainingBatch 10 --model_type 'CNN'
python ./train_model.py --height 128 --width 128 --intial_snap 1 --mid_channels 16 --deltaT 0.1 --trainingBatch 50 --testBatch 1 --type 'CN' --epsilon 0.01 --TotalSnaps 10 --TotalTrainingBatch 10 --model_type 'CNN'


